cd ../..

gpus=$1

for weight in 5
do
python -m trainer.IntplContrastiveTrainer fit \
    --config configs/model/CRNN_contrastive.yaml \
    --config configs/dataset/desed_with_id_od.yaml \
    --config configs/trainer/default.yaml \
    --config configs/experiment/intplcontrastive.yaml \
    --model.exp_name="3_unified_usage+Y800M/in-out-domain/+160h/+wCL_${weight}" \
    --data.init_args.batch_sizes=[12,12,12,0,192] \
    --model.cl_weight=${weight} \
    --model.cl_decay=True \
    --trainer.devices=${gpus} 
done


